from selenium import webdriver

# 启动Edge浏览器
driver = webdriver.Chrome()

# 打开页面
driver.get("file:///F:/TestProject/register.html")

# 使用get_screenshot_as_file()方法保存截图
driver.get_screenshot_as_file("screenshot_file.png")

# 使用save_screenshot()方法保存截图
driver.save_screenshot("screenshot_save.png")

# 使用get_screenshot_as_base64()方法获取Base64编码的截图
base64_screenshot = driver.get_screenshot_as_base64()
print("Base64截图：", base64_screenshot)

# 使用get_screenshot_as_png()方法获取PNG格式的截图
png_screenshot = driver.get_screenshot_as_png()
print("PNG截图：", png_screenshot)

# 关闭浏览器
driver.quit()